

package com.hazelcast.internal.management;

import com.hazelcast.internal.management.operation.*;
import com.hazelcast.internal.serialization.DataSerializerHook;
import com.hazelcast.internal.serialization.impl.ArrayDataSerializableFactory;
import com.hazelcast.internal.serialization.impl.FactoryIdHelper;
import com.hazelcast.nio.serialization.DataSerializableFactory;
import com.hazelcast.nio.serialization.IdentifiedDataSerializable;

import java.util.function.Supplier;

import static com.hazelcast.internal.serialization.impl.FactoryIdHelper.MANAGEMENT_DS_FACTORY;
import static com.hazelcast.internal.serialization.impl.FactoryIdHelper.MANAGEMENT_DS_FACTORY_ID;

public class ManagementDataSerializerHook implements DataSerializerHook {

    public static final int F_ID = FactoryIdHelper.getFactoryId(MANAGEMENT_DS_FACTORY, MANAGEMENT_DS_FACTORY_ID);

    public static final int UPDATE_MAP_CONFIG = 1;
    public static final int SET_LICENSE = 2;
    public static final int CHANGE_CLUSTER_STATE = 3;
    public static final int UPDATE_PERMISSION_CONFIG_OPERATION = 4;
    public static final int RELOAD_CONFIG_OPERATION = 5;
    public static final int UPDATE_CONFIG_OPERATION = 6;
    public static final int UPDATE_TCP_IP_MEMBER_LIST_OPERATION = 7;

    private static final int LEN = UPDATE_TCP_IP_MEMBER_LIST_OPERATION + 1;

    @Override
    public int getFactoryId() {
        return F_ID;
    }

    @Override
    @SuppressWarnings("unchecked")
    public DataSerializableFactory createFactory() {
        Supplier<IdentifiedDataSerializable>[] constructors = new Supplier[LEN];

        constructors[UPDATE_MAP_CONFIG] = UpdateMapConfigOperation::new;
        constructors[SET_LICENSE] = SetLicenseOperation::new;
        constructors[CHANGE_CLUSTER_STATE] = ChangeClusterStateOperation::new;
        constructors[UPDATE_PERMISSION_CONFIG_OPERATION] = UpdatePermissionConfigOperation::new;
        constructors[RELOAD_CONFIG_OPERATION] = ReloadConfigOperation::new;
        constructors[UPDATE_CONFIG_OPERATION] = UpdateConfigOperation::new;
        constructors[UPDATE_TCP_IP_MEMBER_LIST_OPERATION] = UpdateTcpIpMemberListOperation::new;

        return new ArrayDataSerializableFactory(constructors);
    }
}
